# -*- coding: utf-8 -*-
#为观察中值滤波的降噪效果，向灰度图片加入椒盐噪声，并进行3*3的中值滤波器进行中值滤波

from scipy import ndimage
from skimage import data,util
from matplotlib import pyplot as plt

#image为原始图像
img=data.astronaut()[:,:,0]#加载宇航员图片,选取红色通道

#对图像加入椒盐噪声,seed随机种子
noise_img=util.random_noise(img,mode='s&p',seed=None,clip=True)
#plt.imshow(noise_img)

#中值滤波器
n=3
new_img=ndimage.median_filter(noise_img,(n,n))
#plt.imshow(new_img)

#显示图像
plt.figure()
plt.imshow(img,cmap='gray')#显示原始图像
plt.figure()
plt.imshow(noise_img,cmap='gray')#显示加载结果

plt.figure()
plt.imshow(new_img,cmap='gray')#显示降噪哦图片
